package org.pombo.gtb;

import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import org.pombo.gtm.PayoffNode;
import org.pombo.gtm.Player;

public class PayoffNodeBuilder extends Builder<PayoffNode> {

	@Override
	public void clear() {
		this.entity = mock(PayoffNode.class);
	}

	public PayoffNodeBuilder withPayoff(double payoff, Player player) {
		when(this.entity.getPayoff(player)).thenReturn(payoff);
		return this;
	}

}
